tensor([[[0.3000, 0.3000],
         [0.3000, 0.3000]],

        [[0.3300, 0.3300],
         [0.3300, 0.3300]],

        [[0.3700, 0.3700],
         [0.3700, 0.3700]]], grad_fn=<RoundBackward1>)